TensorFlow基础笔记(15) 编译TensorFlow.so,提供给C++平台调用
参考
http://blog.csdn.net/rockingdingo/article/details/75452711
https://www.cnblogs.com/hrlnw/p/7007648.html
https://www.cnblogs.com/hrlnw/p/7383951.html
1.下载tensorflow源码
git clone https://github.com/tensorflow/tensorflow
2.安装bazel
sudo apt-get update && sudo apt-get install bazel
升级bazel
sudo apt-get upgrade bazel
使用bazel编译出现最低版本要求 0.4.5, 而本身的版本为0.11.1的时候,
将 最低版本要求 0.4.5文件中的0.4.5修改为0.0.0
3.编译源码,生成so库
## 进入tensoflow源码根目录后编译
# 编译生成.so文件, 编译C++ API的库 (建议)
bazel build //tensorflow:libtensorflow_cc.so
# 也可以选择,编译C API的库
bazel build //tensorflow:libtensorflow.so
编译出现问题,建议Git clone最新的tensoflow代码 重新编译
编译完成,可以看到编译生成的文件路径
tensorflow/cc/gradients/data_flow_grad.cc:151:1: note: in expansion of macro 'REGISTER_GRADIENT_OP'
REGISTER_GRADIENT_OP("ParallelDynamicStitch", DynamicStitchGrad);
^
Target //tensorflow:libtensorflow_cc.so up-to-date:
bazel-bin/tensorflow/libtensorflow_cc.so
INFO: Elapsed time: 1051.314s, Critical Path: 74.06s
INFO: Build completed successfully, 2355 total actions
4 编译完成,配准调用环境
在等待30多分钟后, 如果编译成功,在tensorflow根目录下出现 bazel-bin, bazel-genfiles 等文件夹,
按顺序执行以下命令将对应的libtensorflow_cc.so文件和其他文件拷贝进入 /usr/local/lib/ 目录
mkdir /usr/local/include/tf
cp -r bazel-genfiles/ /usr/local/include/tf/
cp -r tensorflow /usr/local/include/tf/
cp -r third_party /usr/local/include/tf/
cp -r bazel-bin/tensorflow/libtensorflow_cc.so /usr/local/lib/
这一步完成后,我们就准备好了libtensorflow_cc.so文件等,后面在自己的C++编译环境和代码目录下编译时链接这些库即可。
5 c++调用libtensorflow_cc.so
安装protobuf出现了问题
+ autoreconf -f -i -Wall,no-obsolete
configure.ac:30: error: possibly undefined macro: AC_PROG_LIBTOOL
If this token and others are legitimate, please use m4_pattern_allow.
See the Autoconf documentation.
autoreconf: /usr/bin/autoconf failed with exit status: 1
网上的方案
https://superuser.com/questions/565988/autoconf-libtool-and-an-undefined-ac-prog-libtool
已经安装了 libtool
sudo apt-get install libtool (失败)
可以解决的方法
You need to reinstall it in order to fix the error so follow these steps :
1] Remove current libtool if installed: sudo apt-get purge libtool
2] Download it from official website https://www.gnu.org/software/libtool/
3] Untar it: tar -xzvf "name of the tar_file"
4] Enter folder and type: ./configure && make
5] Install it: sudo make install
And you are done, error should be fixed !
安装步骤
git clone https://github.com/google/protobuf.git
(0)./autogen.sh
(1)./configure --prefix=/usr/local/protobuf
(2)make
(3)make check
(4)make install
注意:
安装成功后,将它的bin和lib目录分别加入到PATH和LD_LIBRARY_PATH环境变量,以方便直接调用。
设置环境变量过程:编辑/etc/profile,在文件末尾添加:
注意:这里添加的路径不同版本protobuf可能不同,需要根据提示确认路径,否则会出现root用户和普通用户版本不同的问题。
export PATH=$PATH:/usr/local/protobuf/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/protobuf/lib
此时最好重启一次。
查看protobuf版本的命令为:protoc --version,如果普通用户无法查看,或者与sudo protoc --version查看的版本不同,就是出现了上述路径添加错误的情况。